torch==1.6.0
transformers==4.2.1
sentencepiece
h5py
wandb
tqdm
numpy
pandas
matplotlib
pyyaml
sacrebleu
torchvision==0.7.0
wget
